from typing import List
from pydantic import BaseModel
from enum import Enum


class PoetryType(str, Enum):
    TangPoetry = "tang_poetry"
    SongLyrics = "song_lyrics"


class Poetry(BaseModel):
    """
    Represents the data structure of a Chinese Poetry.
    """

    type: PoetryType
    title: str
    author: str
    dynasty: str
    content: str
    translation: str
    comment: str
    appreciation: str
    creative_background: str
    brief_analysis: str
    
class PoetryTypeCN(str, Enum):
    唐诗 = "唐诗"
    宋词 = "宋词"
    
class PoetryCN(BaseModel):
    """
    Represents the data structure of a Chinese Poetry.
    """

    类型: PoetryTypeCN
    标题: str
    作者: str
    朝代: str
    内容: str
    译文及注释: str
    赏析: str
    创作背景: str
    简析: str
